package com.bestcem.xm.txtanls.dao.impl.mysql;

import com.bestcem.xm.txtanls.dao.TrainModelDao;
import com.bestcem.xm.txtanls.dao.mapper.TrainModelMapper;
import com.bestcem.xm.txtanls.entity.mysql.TrainModelDo;
import org.springframework.stereotype.Repository;

import javax.annotation.Resource;
import java.util.List;

/**
 * 训练模型
 */
@Repository
public class TrainModelDaoImpl implements TrainModelDao {
    @Resource
    private TrainModelMapper trainModelMapper;

    @Override
    public List<TrainModelDo> listTrainModel() {
        return trainModelMapper.listTrainModel();
    }

    @Override
    public TrainModelDo getByCode(Integer code) {
        return trainModelMapper.getByCode(code);
    }

    @Override
    public TrainModelDo getById(String trainModelId) {
        return trainModelMapper.getById(trainModelId);
    }
}
